export const llmAllDataKey = [
  'model_uid',
  'model_name',
  'model_type',
  'model_engine',
  'model_format',
  'model_size_in_billions',
  'quantization',
  'n_worker',
  'n_gpu',
  'n_gpu_layers',
  'replica',
  'request_limits',
  'worker_ip',
  'gpu_idx',
  'download_hub',
  'model_path',
  'reasoning_content',
  'gguf_quantization',
  'gguf_model_path',
  'cpu_offload',
  'peft_model_config',
]

export const additionalParameterTipList = {
  'transformers': ['torch_dtype', 'device'],
  'llama.cpp': ['n_ctx', 'use_mmap', 'use_mlock'],
  'vllm': [
    'block_size',
    'gpu_memory_utilization',
    'max_num_seqs',
    'max_model_len',
    'guided_decoding_backend',
    'scheduling_policy',
    'tensor_parallel_size',
    'pipeline_parallel_size',
    'enable_prefix_caching',
    'enable_chunked_prefill',
    'enforce_eager',
    'cpu_offload_gb',
    'disable_custom_all_reduce',
    'limit_mm_per_prompt',
    'model_quantization',
  ],
  'sglang': [
    'mem_fraction_static',
    'attention_reduce_in_fp32',
    'tp_size',
    'dp_size',
    'chunked_prefill_size',
    'cpu_offload_gb',
    'enable_dp_attention',
    'enable_ep_moe',
  ],
  'mlx': ['cache_limit_gb', 'max_kv_size'],
}

export const featureModels = [
  {
    type: 'llm',
    feature_models: [
      'deepseek-v3',
      'deepseek-r1',
      'deepseek-r1-distill-qwen',
      'deepseek-r1-distill-llama',
      'qwen2.5-instruct',
      'qwen2.5-vl-instruct',
      'qwen2.5-coder-instruct',
      'QwQ-32B',
      'llama-3.1-instruct',
      'gemma-3-it',
    ],
  },
  {
    type: 'embedding',
    feature_models: [
      'bge-large-zh-v1.5',
      'bge-large-en-v1.5',
      'bge-m3',
      'gte-Qwen2',
      'jina-embeddings-v3',
    ],
  },
  {
    type: 'rerank',
    feature_models: [],
  },
  {
    type: 'image',
    feature_models: [
      'FLUX.1-dev',
      'FLUX.1-schnell',
      'sd3.5-large',
      'HunyuanDiT-v1.2',
      'sd3.5-medium',
    ],
  },
  {
    type: 'audio',
    feature_models: [
      'CosyVoice2-0.5B',
      'FishSpeech-1.5',
      'F5-TTS',
      'ChatTTS',
      'SenseVoiceSmall',
      'whisper-large-v3',
    ],
  },
  {
    type: 'video',
    feature_models: [],
  },
]
